package org.example.graph_theory;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Pacific_Atlantic_water_flow_problem {
    public static void main(String[] args) {
        //太平洋大西洋水流问题

        //有一个 m × n 的矩形岛屿，与 太平洋 和 大西洋 相邻。 “太平洋” 处于大陆的左边界和上边界，而 “大西洋” 处于大陆的右边界和下边界。
        //这个岛被分割成一个由若干方形单元格组成的网格。给定一个 m x n 的整数矩阵 heights ， heights[r][c] 表示坐标 (r, c) 上单元格 高于海平面的高度 。
        //岛上雨水较多，如果相邻单元格的高度 小于或等于 当前单元格的高度，雨水可以直接向北、南、东、西流向相邻单元格。水可以从海洋附近的任何单元格流入海洋。
        //返回网格坐标 result 的 2D 列表 ，其中 result[i] = [ri, ci] 表示雨水从单元格 (ri, ci) 流动 既可流向太平洋也可流向大西洋 。
//        int[][] heights = {{1, 2, 2, 3, 5}, {3, 2, 3, 4, 4}, {2, 4, 5, 3, 1}, {6, 7, 1, 4, 5}, {5, 1, 1, 2, 4}};
        int[][] heights = {{1}};
        List<List<Integer>> lists = pacificAtlantic(heights);
        System.out.println(lists);
    }

    public static List<List<Integer>> pacificAtlantic(int[][] heights) {
        ArrayList<List<Integer>> listArrayList = new ArrayList<>();
        String[][] nums = new String[heights.length][heights[0].length];
        for (int i = 0; i < heights[0].length; i++) {
            dfs(heights, 0, i, nums);
        }
        for (int i = 0; i < heights.length; i++) {
            dfs(heights, i, 0, nums);
        }

        for (int i = 0; i < heights[heights.length - 1].length ; i++) {
            dfs2(heights, heights.length - 1, i, nums);
        }
        for (int i = 0; i < heights.length; i++) {
            dfs2(heights, i, heights[0].length - 1, nums);
        }

        for (int i = 0; i < nums.length; i++) {
            for (int j = 0; j < nums[i].length; j++) {
                if ("xy".equals(nums[i][j])) {
                    ArrayList<Integer> list = new ArrayList<>();
                    list.add(i);
                    list.add(j);
                    listArrayList.add(list);
                }
            }
        }
        System.out.println(Arrays.deepToString(nums));
        return listArrayList;
    }

    public static void dfs(int[][] heights, int i, int j, String[][] nums) {
        if (i < 0 || j < 0 || i >= heights.length || j >= heights[i].length || "x".equals(nums[i][j])) return;
        nums[i][j] = "x";
        if (i - 1 >= 0 && heights[i][j] <= heights[i - 1][j]) dfs(heights, i - 1, j, nums);
        if (i + 1 < heights.length && heights[i][j] <= heights[i + 1][j]) dfs(heights, i + 1, j, nums);
        if (j - 1 >= 0 && heights[i][j] <= heights[i][j - 1]) dfs(heights, i, j - 1, nums);
        if (j + 1 < heights[i].length && heights[i][j] <= heights[i][j + 1]) dfs(heights, i, j + 1, nums);
    }

    public static void dfs2(int[][] heights, int i, int j, String[][] nums) {
        if (i < 0 || j < 0 || i >= heights.length || j >= heights[i].length || "y".equals(nums[i][j]) || "xy".equals(nums[i][j])) return;
        if ("x".equals(nums[i][j])) {
            nums[i][j] += "y";
        } else {
            nums[i][j] = "y";
        }
        if (i - 1 >= 0 && heights[i][j] <= heights[i - 1][j]) dfs2(heights, i - 1, j, nums);
        if (i + 1 < heights.length && heights[i][j] <= heights[i + 1][j]) dfs2(heights, i + 1, j, nums);
        if (j - 1 >= 0 && heights[i][j] <= heights[i][j - 1]) dfs2(heights, i, j - 1, nums);
        if (j + 1 < heights[i].length && heights[i][j] <= heights[i][j + 1]) dfs2(heights, i, j + 1, nums);
    }
}
